package com.ruoyi.central.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.github.pagehelper.page.PageMethod;
import com.ruoyi.cache.common.CacheConstants;
import com.ruoyi.common.enums.common.TaskStatusEnum;
import com.ruoyi.common.utils.CollectionUtils;
import com.ruoyi.common.utils.PageHelperUtils;
import lombok.extern.slf4j.Slf4j;
import com.baomidou.mybatisplus.core.toolkit.support.SFunction;
import com.baomidou.mybatisplus.extension.toolkit.SqlHelper;
import org.springframework.cache.annotation.CacheConfig;
import org.springframework.cache.annotation.CacheEvict;
import org.springframework.cache.annotation.Cacheable;
import java.io.Serializable;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
import org.springframework.stereotype.Service;
import com.ruoyi.common.core.service.ServiceImpl;
import com.ruoyi.central.mapper.TopCentralProcessStepRunMapper;
import com.ruoyi.central.domain.TopCentralProcessStepRun;
import com.ruoyi.central.service.ITopCentralProcessStepRunService;
import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.common.enums.common.State;

import javax.validation.constraints.NotEmpty;
import javax.validation.constraints.NotNull;
import java.util.Collection;
import java.util.ArrayList;

/**
 * 执行工序Service业务层处理
 *
 * @author SunLuQuan
 * @date 2022-04-26
 */
@Service
@Slf4j
@CacheConfig(cacheNames = CacheConstants.EhcacheName.CONTINUAL_CACHE_NAME)
public class TopCentralProcessStepRunServiceImpl extends ServiceImpl<TopCentralProcessStepRunMapper,TopCentralProcessStepRun> implements ITopCentralProcessStepRunService
{
    /**
     * 查询执行工序
     * 
     * @param id 执行工序主键
     * @param columns 查询指定的列
     * @return 执行工序
     */
    @Override
    public TopCentralProcessStepRun selectById(Integer id, SFunction<TopCentralProcessStepRun,?> ... columns){

        if(StringUtils.isNotEmpty(columns)){
            return selectOne(lambdaQuery().select(columns).eq(TopCentralProcessStepRun::getId,id));
        }
        return super.selectById(id);
    }

    /**
     * 查询执行工序
         * 注意：只获取状态信息为可用的数据
     * 
     * @param id 执行工序主键
     * @param columns 查询指定的列
     * @return 执行工序
     */
    @Override
    public TopCentralProcessStepRun selectUsableById(Integer id, SFunction<TopCentralProcessStepRun,?> ... columns)
    {
            LambdaQueryWrapper<TopCentralProcessStepRun> lambdaQueryWrapper = lambdaQuery().eq(TopCentralProcessStepRun::getId,id);
            buildUsableState(lambdaQueryWrapper);
            if(StringUtils.isNotEmpty(columns)){
                lambdaQueryWrapper.select(columns);
            }
            return selectOne(lambdaQueryWrapper);
    }

    /**
     * 查询执行工序列表
     *
     * @param topCentralProcessStepRun 执行工序
     * @return 执行工序
     */
    @Override
    public List<TopCentralProcessStepRun> selectList(TopCentralProcessStepRun topCentralProcessStepRun)
    {
        return getBaseMapper().selectTopCentralProcessStepRunList(topCentralProcessStepRun);
    }

    /**
     * 根据工艺执行编号 获取工艺执行编号下所有的执行工序 除了 未开始的
     * @param artProcessRunId 执行工艺编号
     * @return 工艺执行编号下所有的执行工序 除了 未开始的 执行工序列表
     */
    @Override
    public List<TopCentralProcessStepRun> getTimeLineByArtProcessRunId(Integer artProcessRunId) {

        LambdaQueryWrapper<TopCentralProcessStepRun> queryWrapper = customLambdaQuery();

        queryWrapper.eq(TopCentralProcessStepRun::getArtProcessRunId,artProcessRunId);
        queryWrapper.ne(TopCentralProcessStepRun::getTaskStatus,TaskStatusEnum.NOT_START);

        queryWrapper.orderByAsc(TopCentralProcessStepRun::getWeight);

        return getBaseMapper().selectList(queryWrapper);
    }

    /**
     * 根据工艺执行编号 获取执行中的工序任务以及 执行中工序任务的下一个
     * @param artProcessRunId 工艺执行编号
     * @return 获取执行中的工序任务以及 执行中工序任务的下一个
     */
    @Override
    public List<TopCentralProcessStepRun> selectByArtProcessRunId(@NotNull Integer artProcessRunId) {

        LambdaQueryWrapper<TopCentralProcessStepRun> queryWrapper = customLambdaQuery();
        queryWrapper.eq(TopCentralProcessStepRun::getArtProcessRunId,artProcessRunId);
        queryWrapper.ne(TopCentralProcessStepRun::getTaskStatus,TaskStatusEnum.EXECUTED_FINISH.getValue());

        queryWrapper.orderByAsc(TopCentralProcessStepRun::getWeight);

        PageHelperUtils.startPage(1,2);

        return getBaseMapper().selectList(queryWrapper);
    }

    /**
     * 根据工艺执行编号列表  获取工艺执行下所有的 绑定的 工序执行对象
     * @param artProcessRunIds 工艺执行编号集合
     * @return k工艺执行编号 v 正在执行中的工序执行对象
     */
    @Override
    public Map<Integer, TopCentralProcessStepRun> selectByArtProcessRunIds(@NotEmpty Collection<Integer> artProcessRunIds) {

        LambdaQueryWrapper<TopCentralProcessStepRun> queryWrapper = customLambdaQuery()
                .select(TopCentralProcessStepRun::getArtProcessRunId,TopCentralProcessStepRun::getName,TopCentralProcessStepRun::getTaskStatus)
                .in(TopCentralProcessStepRun::getTaskStatus, TaskStatusEnum.getActiveTaskStatusEnumList());

        if(artProcessRunIds.size()==1){
            queryWrapper.eq(TopCentralProcessStepRun::getArtProcessRunId, CollectionUtils.getFirst(artProcessRunIds));
        }else{
            queryWrapper.in(TopCentralProcessStepRun::getArtProcessRunId, artProcessRunIds);
        }

        return CollectionUtils.toMap(getBaseMapper().selectList(queryWrapper),TopCentralProcessStepRun::getArtProcessRunId);
    }

    /**
     * 获取所有的执行工序数据
     * 
     * @return Map<Integer,TopCentralProcessStepRun> 集合
     */
    @Cacheable
    @Override
    public Map<Integer,TopCentralProcessStepRun> selectAllMap(){

        return getBaseMapper().selectList(customLambdaQuery())
                .stream().collect(Collectors.toMap(TopCentralProcessStepRun::getId,v->v));
    }

    /**
     * 新增执行工序
     *
     * @param topCentralProcessStepRun 执行工序
     * @return 结果
     */
    @CacheEvict(keyGenerator=CacheConstants.KeyGeneratorName.EHCACHE_KEY_GENERATOR_EVICT)
    @Override
    public boolean insert(TopCentralProcessStepRun topCentralProcessStepRun)
    {
        return super.insert(topCentralProcessStepRun);
    }

    /**
     * 新增（批量）执行工序
     *
     * @param topCentralProcessStepRunList 实体对象集合
     */
    @CacheEvict(keyGenerator=CacheConstants.KeyGeneratorName.EHCACHE_KEY_GENERATOR_EVICT)
    @Override
    public boolean insertBatch(Collection<TopCentralProcessStepRun> topCentralProcessStepRunList, int batchSize) {
        return super.insertBatch(topCentralProcessStepRunList, batchSize);
    }

    /**
     * 删除 执行工序 根据编号
     * @param id
     * @return true 删除成功 false 删除失败
     */
    @CacheEvict(keyGenerator=CacheConstants.KeyGeneratorName.EHCACHE_KEY_GENERATOR_EVICT)
    @Override
    public boolean deleteById(Serializable id) {
        return super.deleteById(id);
    }

    /**
     * 批量删除 执行工序 根据 编号集合
     * @param ids
     * @return true 删除成功 false 删除失败
     */
    @CacheEvict(keyGenerator=CacheConstants.KeyGeneratorName.EHCACHE_KEY_GENERATOR_EVICT)
    @Override
    public boolean deleteByIds(Collection<? extends Serializable> ids) {
        return super.deleteByIds(ids);
    }

    /**
     * 批量修改 执行工序 根据唯一编号
     * @param topCentralProcessStepRunList
     * @param batchSize  更新批次数量
     * @return true 修改成功 false 修改失败
     */
    @CacheEvict(keyGenerator=CacheConstants.KeyGeneratorName.EHCACHE_KEY_GENERATOR_EVICT)
    @Override
    public boolean updateBatchById(Collection<TopCentralProcessStepRun> topCentralProcessStepRunList, int batchSize) {
        return super.updateBatchById(topCentralProcessStepRunList, batchSize);
    }

    /**
     * 修改 执行工序
     * @param  topCentralProcessStepRun
     * @param updateWrapper
     * @return
     */
    @CacheEvict(keyGenerator=CacheConstants.KeyGeneratorName.EHCACHE_KEY_GENERATOR_EVICT)
    @Override
    public boolean update(TopCentralProcessStepRun topCentralProcessStepRun, LambdaUpdateWrapper<TopCentralProcessStepRun> updateWrapper) {
        return super.update(topCentralProcessStepRun,updateWrapper);
    }

    /**
     * 修改执行工序
     *
     * @param topCentralProcessStepRun 执行工序
     * @return 结果
     */
    @CacheEvict(keyGenerator=CacheConstants.KeyGeneratorName.EHCACHE_KEY_GENERATOR_EVICT)
    @Override
    public boolean updateById(TopCentralProcessStepRun topCentralProcessStepRun)
    {
        return super.updateById(topCentralProcessStepRun);
    }

    /**
     * 修改执行工序状态 根据 id 执行工序
     * 
     * @param id 执行工序主键
     * @param state 数据状态
     * @return 结果
     */
    @CacheEvict(keyGenerator=CacheConstants.KeyGeneratorName.EHCACHE_KEY_GENERATOR_EVICT)
    @Override
    public boolean updateStateById(Integer id, State state){

        return updateById(buildTopCentralProcessStepRunForUpdateState(id,state));
    }

    /**
     * 修改多个执行工序状态 根据 id 执行工序
     * 
     * @param ids 执行工序主键
     * @param state 数据状态
     * @return 结果
     */
    @CacheEvict(keyGenerator=CacheConstants.KeyGeneratorName.EHCACHE_KEY_GENERATOR_EVICT)
    @Override
    public boolean updateStateByIds(Collection<Integer> ids, State state){

        List<TopCentralProcessStepRun> updateTopCentralProcessStepRunList = new ArrayList(ids.size());

            ids.forEach(id->updateTopCentralProcessStepRunList.add(buildTopCentralProcessStepRunForUpdateState(id,state)));

        return updateBatchById(updateTopCentralProcessStepRunList);
    }

    /**
     * 构建修改TopCentralProcessStepRun 的 修改类
     * @param id 执行工序主键
     * @param state 数据状态
     * @return TopCentralProcessStepRun构建的修改模型
     */
    private TopCentralProcessStepRun buildTopCentralProcessStepRunForUpdateState(Integer id, State state){

        TopCentralProcessStepRun topCentralProcessStepRun = new TopCentralProcessStepRun();
        topCentralProcessStepRun.setState(state);
        topCentralProcessStepRun.setId(id);

        return topCentralProcessStepRun;
    }

}
